import pandas as pd
# 内蒙古中西部三市人口数据示例
data = {
    '城市': ['呼和浩特', '包头', '鄂尔多斯'],
    '人口(万)': [310, 280, 210],
    '区域': ['中部', '中部', '西部']
}
df = pd.DataFrame(data)

# 下面的代码为Ch08-20.1部分
# 使用 map 对 区域 列进行编码
# region_map = {'中部': 0, '西部': 1}
# df['地区编码'] = df['区域'].map(region_map)
# print(df)

# 下面的代码为Ch08-20.2部分，运行时注释Ch08-20.1的代码
# 基于Ch08-20.1创建的Dataframe模拟数据[df]
# 计算人口密度（假设面积为固定值）
area = {'呼和浩特': 17200, '包头': 27239, '鄂尔多斯': 86752}
df['密度'] = df.apply(lambda row: row['人口(万)'] / area[row['城市']], axis=1)
print(df)
